import re


# 定义一个字典来存储常见问题和对应的回答
responses = {
    "你好": ["你好呀！有什么我可以帮忙的？", "哈喽，今天过得怎么样？"],
    "你是谁": ["我是一个聊天机器人呀", "我叫小助手，随时为你服务"],
    "再见": ["再见啦，祝你有个美好的一天！", "拜拜，下次再聊~"]
}


def simple_chatbot(user_input):
    # 检查是否是数学计算
    match = re.search(r'(\d+)\s*([+\-*/%])\s*(\d+)', user_input)
    if match:
        num1 = float(match.group(1))
        operator = match.group(2)
        num2 = float(match.group(3))
        if operator == '+':
            result = num1 + num2
        elif operator == '-':
            result = num1 - num2
        elif operator == '*':
            result = num1 * num2
        elif operator == '/':
            if num2 != 0:
                result = num1 / num2
            else:
                return "除数不能为零"
        elif operator == '%':
            result = num1 % num2
        return f"计算结果是: {result}"

    for key in responses.keys():
        if key in user_input:
            import random
            return random.choice(responses[key])
    return "我不太明白你的意思，可以换个说法吗？"


while True:
    user_input = input("你: ").strip()
    if user_input.lower() == '退出':
        break
    response = simple_chatbot(user_input)
    print("聊天机器人: ", response)
